Making a command line prompt program in Python